package com;

import javax.crypto.Cipher;
import javax.crypto.spec.SecretKeySpec;

public class Aes {

	public static void main(String[] args) throws Exception {

	}

	// 加密
	public static String Encrypt(String sSrc, String sKey) throws Exception {
		if (sKey == null) {
			System.out.print("Key为空null");
			return null;
		}
		sKey = MD5.getStrMd5(sKey);
		byte[] raw = sKey.getBytes("utf-8");
		SecretKeySpec skeySpec = new SecretKeySpec(raw, "AES");
		Cipher cipher = Cipher.getInstance("AES/ECB/PKCS5Padding");// "算法/模式/补码方式"
		cipher.init(Cipher.ENCRYPT_MODE, skeySpec);
		byte[] encrypted = cipher.doFinal(sSrc.getBytes("utf-8"));

		return Base64.encode(encrypted).replace("/", "_").replace("+", "-").replace("=", "$");// 此处使用BASE64做转码功能，同时能起到2次加密的作用。
	}

	// 解密
	public static String Decrypt(String sSrc, String sKey) throws Exception {

		if (sSrc.length() % 4 == 2) {
			sSrc += "==";
		} else if (sSrc.length() % 4 == 3) {
			sSrc += "=";
		}
		sSrc = sSrc.replace("_", "/").replace("-", "+").replace("$", "=");

		// 判断Key是否正确
		if (sKey == null) {
			System.out.print("Key为空null");
			return null;
		}
		sKey = MD5.getStrMd5(sKey);
		byte[] raw = sKey.getBytes("utf-8");
		SecretKeySpec skeySpec = new SecretKeySpec(raw, "AES");
		Cipher cipher = Cipher.getInstance("AES/ECB/PKCS5Padding");
		cipher.init(Cipher.DECRYPT_MODE, skeySpec);
		byte[] encrypted1 = Base64.decode(sSrc);// 先用base64解密
		byte[] original = cipher.doFinal(encrypted1);
		String originalString = new String(original, "utf-8");
		return originalString;
	}
}
